Hierarchical Sparse Attention Done Right: Toward Infinite Context Modeling
分层稀疏注意力HiLS:端到端分块选择实现无限上下文建模
🧭 分层稀疏注意力 · 超长上下文外推与高效推理
Tencent HY Team · ShanghaiTech University · The Hong Kong University of Science and Technology · University of California, San Diego
提出HiLS Attention,端到端学习分块选择的分层稀疏注意力,推理加速且外推超64倍训练长度
📌 工作参考价值(4 点)
- 在电商推荐的长用户行为序列建模中,可借鉴HiLS的分块检索与端到端选择机制,自动关注关键历史块,突破全注意力计算瓶颈,同时保持预测性能。
- 对于RAG或Agent长上下文场景,分层地标(landmark)压缩文档/对话历史,实现高效检索式注意力计算,既能大幅降低KV缓存,又能外推到极长序列,适合在线长对话或长文档问答。
- HiLS提供全注意力模型的轻量级继续预训练转换方案,已有模型(如推荐系统、对话模型)只需少量训练即可获得超长上下文能力和推理加速,便于业务快速升级。
- 推理时只缓存被选中的分块KV,降低解码内存和延迟(预填充加速9.3x,解码加速15.7x),适合在线服务对长序列实时响应的要求。